function [display_var_balanced display_var_schoolshort] = robustness_imputation(store_suffix)



basedata=load('../../01_data/05_temp/model/alloc_teach_est_princ_est_restr_imputed.mat');
balanced_restr=basedata.balanced;
%%%
% restricted timing
%%%
base=['alloc_teach_est_princ_est_restr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_restr=basedata.balanced;
school_short_restr=basedata.school_short;

%%%
% restricted timing, noisy
%%%
base=['alloc_teach_est_princ_noisy_restr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_noisy=basedata.balanced;
school_short_noisy=basedata.school_short;


%%%
% unrestricted timing
%%%
base=['alloc_teach_est_princ_est_unrestr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_unrestr=basedata.balanced;
school_short_unrestr=basedata.school_short;

%%%
% restricted timing, principal max va 
%%%
base=['alloc_teach_est_princ_va_restr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_va=basedata.balanced;
school_short_va=basedata.school_short;


%%%
% restricted timing, teach max ndis
%%%
base=['alloc_teach_ndis_princ_est_restr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_ndis=basedata.balanced;
school_short_ndis=basedata.school_short;



%%%
% restricted timing, teach max ndis, principal max va
%%%
base=['alloc_teach_ndis_princ_va_restr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_both=basedata.balanced;
school_short_both=basedata.school_short;


%%%  
%  6. status quo, principal max VA, unrestricted choice sets, also compute first best 
%%%

base=['alloc_teach_est_princ_va_unrestr_',store_suffix,'.mat'];
basefilename=join(base);
folder='../../01_data/05_temp/model/';
savefile=fullfile(folder, basefilename);


basedata=open(savefile);

balanced_fb=basedata.balanced;
school_short_fb=basedata.school_short;


%%%
% collect things
%%%%

% status_quo = balanced_restr,  school_short_restr
% all_options = balanced_unrestr  school_short_unrestr
% principal_va = balanced_va school_short_va 
% teach_fracdis_princ_est = balanced_ndis school_short_ndis 
% teach_fracdis_princ_va = balanced_both  school_short_both
% first_best = balanced_fb  school_short_fb

display_var_balanced= [balanced_restr(1,1) balanced_unrestr(1,1) balanced_va(1,1) balanced_ndis(1,1) balanced_both(1,1) balanced_fb(1,1)];
display_var_schoolshort= [school_short_restr(1,1) school_short_unrestr(1,1) school_short_va(1,1) school_short_ndis(1,1) school_short_both(1,1) school_short_fb(1,1)];

end 

